User Defined Function in python